mapping = {'inception_v3':['https://tfhub.dev/google/imagenet/inception_v3/feature_vector/1', 2048],
           'resnet_v2_50':['https://tfhub.dev/google/imagenet/resnet_v2_50/feature_vector/1', 2048],
           'mobilenet_v2_140_224':['https://tfhub.dev/google/imagenet/mobilenet_v2_140_224/feature_vector/2', 1792],
           'mobilenet_v1_100_224':['https://tfhub.dev/google/imagenet/mobilenet_v1_100_224/feature_vector/1', 1024],
           'mobilenet_v2_100_224':['https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/feature_vector/2', 1280],
           'inception_resnet_v2':['https://tfhub.dev/google/imagenet/inception_resnet_v2/feature_vector/1', 1536],
           'nasnet_large':['https://tfhub.dev/google/imagenet/nasnet_large/feature_vector/1', 4032],
           'nasnet_mobile':['https://tfhub.dev/google/imagenet/nasnet_mobile/feature_vector/1', 1056],
           'pnasnet_large':['https://tfhub.dev/google/imagenet/pnasnet_large/feature_vector/2', 4320],
           'mobilenet_v2_035_128':['https://tfhub.dev/google/imagenet/mobilenet_v2_035_128/feature_vector/2', 1280],
           'mobilenet_v2_100_96':['https://tfhub.dev/google/imagenet/mobilenet_v2_100_96/feature_vector/2', 1280],
           'mobilenet_v1_050_224':['https://tfhub.dev/google/imagenet/mobilenet_v1_050_224/feature_vector/1', 512],
           'inception_v1':['https://tfhub.dev/google/imagenet/inception_v1/feature_vector/1', 1024],
           'resnet_v2_152':['https://tfhub.dev/google/imagenet/resnet_v2_152/feature_vector/1', 2048],
           'mobilenet_v2_035_224':['https://tfhub.dev/google/imagenet/mobilenet_v2_035_224/feature_vector/2', 1280],
           'mobilenet_v2_130_224':['https://tfhub.dev/google/imagenet/mobilenet_v2_130_224/feature_vector/2', 1664],
           'mobilenet_v2_050_224':['https://tfhub.dev/google/imagenet/mobilenet_v2_050_224/feature_vector/2', 1280],
           'mobilenet_v2_100_160':['https://tfhub.dev/google/imagenet/mobilenet_v2_100_160/feature_vector/2', 1280],
           'mobilenet_v2_050_128':['https://tfhub.dev/google/imagenet/mobilenet_v2_050_128/feature_vector/2', 1280],
           'mobilenet_v1_100_128':['https://tfhub.dev/google/imagenet/mobilenet_v1_100_128/feature_vector/1', 1024],
           'mobilenet_v2_075_224':['https://tfhub.dev/google/imagenet/mobilenet_v2_075_224/feature_vector/2', 1280],
           'mobilenet_v2_100_128':['https://tfhub.dev/google/imagenet/mobilenet_v2_100_128/feature_vector/2', 1280],
           'mobilenet_v2_100_192':['https://tfhub.dev/google/imagenet/mobilenet_v2_100_192/feature_vector/2', 1280],
           'mobilenet_v1_025_224':['https://tfhub.dev/google/imagenet/mobilenet_v1_025_224/feature_vector/1', 256],
           'mobilenet_v2_035_96':['https://tfhub.dev/google/imagenet/mobilenet_v2_035_96/feature_vector/2', 1280],
           'mobilenet_v2_075_96':['https://tfhub.dev/google/imagenet/mobilenet_v2_075_96/feature_vector/2', 1280],
           'resnet_v2_101':['https://tfhub.dev/google/imagenet/resnet_v2_101/feature_vector/1', 2048],
           'amoebanet_a_n18_f448':['https://tfhub.dev/google/imagenet/amoebanet_a_n18_f448/feature_vector/1', 7168],
           'mobilenet_v2_050_96':['https://tfhub.dev/google/imagenet/mobilenet_v2_050_96/feature_vector/2', 1280],
           'mobilenet_v2_075_192':['https://tfhub.dev/google/imagenet/mobilenet_v2_075_192/feature_vector/2', 1280],
           'mobilenet_v1_025_128':['https://tfhub.dev/google/imagenet/mobilenet_v1_025_128/feature_vector/1', 256],
           'mobilenet_v1_050_160':['https://tfhub.dev/google/imagenet/mobilenet_v1_050_160/feature_vector/1', 512],
           'mobilenet_v2_075_160':['https://tfhub.dev/google/imagenet/mobilenet_v2_075_160/feature_vector/2', 1280],
           'mobilenet_v2_050_160':['https://tfhub.dev/google/imagenet/mobilenet_v2_050_160/feature_vector/2', 1280],
           'mobilenet_v1_075_160':['https://tfhub.dev/google/imagenet/mobilenet_v1_075_160/feature_vector/1', 768],
           'mobilenet_v1_075_224':['https://tfhub.dev/google/imagenet/mobilenet_v1_075_224/feature_vector/1', 768],
           'mobilenet_v1_050_192':['https://tfhub.dev/google/imagenet/mobilenet_v1_050_192/feature_vector/1', 512],
           'mobilenet_v2_035_192':['https://tfhub.dev/google/imagenet/mobilenet_v2_035_192/feature_vector/2', 1280],
           'mobilenet_v1_075_128':['https://tfhub.dev/google/imagenet/mobilenet_v1_075_128/feature_vector/1', 768],
           'inception_v2':['https://tfhub.dev/google/imagenet/inception_v2/feature_vector/1', 1024],
            'mobilenet_v1_075_192':['https://tfhub.dev/google/imagenet/mobilenet_v1_075_192/feature_vector/1', 768],
            'mobilenet_v1_100_160':['https://tfhub.dev/google/imagenet/mobilenet_v1_100_160/feature_vector/1', 1024],
            'mobilenet_v1_100_192':['https://tfhub.dev/google/imagenet/mobilenet_v1_100_192/feature_vector/1', 1024],
            'mobilenet_v2_075_128':['https://tfhub.dev/google/imagenet/mobilenet_v2_075_128/feature_vector/2', 1280],
            'mobilenet_v1_050_128':['https://tfhub.dev/google/imagenet/mobilenet_v1_050_128/feature_vector/1', 512],
            'resnet_v1_50':['https://tfhub.dev/google/imagenet/resnet_v1_50/feature_vector/1', 2048],
            'mobilenet_v2_035_160':['https://tfhub.dev/google/imagenet/mobilenet_v2_035_160/feature_vector/2', 1280],
            'mobilenet_v2_050_192':['https://tfhub.dev/google/imagenet/mobilenet_v2_050_192/feature_vector/2', 1280],
            'resnet_v1_101':['https://tfhub.dev/google/imagenet/resnet_v1_101/feature_vector/1', 2048],
            'mobilenet_v1_025_160':['https://tfhub.dev/google/imagenet/mobilenet_v1_025_160/feature_vector/1', 256],
            'mobilenet_v1_025_192':['https://tfhub.dev/google/imagenet/mobilenet_v1_025_192/feature_vector/1', 256],
            'resnet_v1_152':['https://tfhub.dev/google/imagenet/resnet_v1_152/feature_vector/1', 2048]
           }